package org.virtualdb.socket.proxy;

/*
 * Copyright 2012 The Netty Project
 *
 * The Netty Project licenses this file to you under the Apache License,
 * version 2.0 (the "License"); you may not use this file except in compliance
 * with the License. You may obtain a copy of the License at:
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations
 * under the License.
 */

import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.ChannelOption;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.nio.NioServerSocketChannel;

public class JSockectProxy {

    private final int localPort;
    private final String remoteHost;
    private final int remotePort;

    public JSockectProxy(int localPort, String remoteHost, int remotePort) {
        this.localPort = localPort;
        this.remoteHost = remoteHost;
        this.remotePort = remotePort;
    }

    public void run() throws Exception {
        System.out.println("Proxying *:" + localPort + " to " + remoteHost + ':' + remotePort + " ...");

        EventLoopGroup bossGroup = new NioEventLoopGroup();
        EventLoopGroup workerGroup = new NioEventLoopGroup();
        try {
            ServerBootstrap b = new ServerBootstrap();
            b.group(bossGroup, workerGroup).channel(NioServerSocketChannel.class)
                    .childHandler(new JSockProxyInitializer(remoteHost, remotePort))
                    .childOption(ChannelOption.AUTO_READ, false).bind(localPort).sync().channel().closeFuture().sync();
        } finally {
            bossGroup.shutdownGracefully();
            workerGroup.shutdownGracefully();
        }
    }

    public static void main(String[] args) throws Exception {
        if (args.length < 3) {
            System.out.println("Usage: java -jar jsock-proxy.jar  <local port> <remote host> <remote port>");
            return;
        }

        String remoteHost = args[1];
        int localPort = Integer.parseInt(args[0]);
        int remotePort = Integer.parseInt(args[2]);
        if (args.length == 4 && "log".equals(args[3])) {
            System.setProperty("do_log", "true");
        }
        new JSockectProxy(localPort, remoteHost, remotePort).run();
    }

}
